package com.ai.leetcode;

/**
 * 买卖股票的最佳时机 II
 * https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-ii/description/?envType=study-plan-v2&envId=top-interview-150
 */
public class Code001_BestTimeToBuyAndSellStock2 {

    public static void main(String[] args) {
        int[] nums = {7,1,5,3,6,4};
        int i = maxProfit(nums);
        System.out.println(i);
    }

    /**
     * {1,4,2}
     * DP动态规划，
     * 第i天只有两种状态，不持有或持有股票，
     * 当天不持有股票的状态可能来自昨天卖出或者昨天也不持有，
     * 同理，当天持有股票的状态可能来自昨天买入或者昨天也持有中，取最后一天的不持有股票状态就是问题的解
     *
     * @param prices
     * @return
     */
    public static int maxProfit(int[] prices) {
        if (prices == null) {
            return 0;
        }
        int n = prices.length;
        int[][] dp = new int[prices.length][2];
        // dp[i][0]表示第i天不持有股票, dp[i][1]表示第i天持有股票
        dp[0][0] = 0;
        dp[0][1] = -prices[0];
        for (int i = 1; i < n; i++) {
            // 当天不持有股票的状态可能来自昨天卖出或者昨天也不持有，
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
            // 当天持有股票的状态可能来自昨天买入或者昨天也持有中
            dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
        }
        return dp[n - 1][0];
    }
}
